"""
Downloading: "https://github.com/ultralytics/yolov5/archive/master.zip" to /home/asuspei/.cache/torch/hub/master.zip
YOLOv5 🚀 2022-10-21 Python-3.7.13 torch-1.10.1+cu111 CUDA:0 (NVIDIA GeForce RTX 2060, 12030MiB)

Downloading https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5s.pt to yolov5s.pt...
100%|██████████| 14.1M/14.1M [00:02<00:00, 6.40MB/s]

Fusing layers...
YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients, 16.4 GFLOPs
Adding AutoShape...

Process finished with exit code 0
"""
import torch
import sys

model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

img_path = '/home/asuspei/large_data/pic/dog_bird.jpg'
argc = len(sys.argv)
if argc >= 2:
    img_path = sys.argv[1]

results = model(img_path)

results.show()
